from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import PromptTemplate

from a0base.base_llm import pop_llm

# 设置本地模型，不使用深度思考
prompt = PromptTemplate.from_template(
    "请回答我的问题：{question}"
)
# 创建字符串输出解析器
parser = StrOutputParser()

# 构建链式调用
chain = prompt | pop_llm | parser

# 执行链式调用
print(chain.invoke({"question": "我叫崔亮，你叫什么?"}))
print(chain.invoke({"question": "你知道我是谁吗?"}))
